def add(a, b):
    return a + b
def sub(a, b):
    return a - b


if __name__ == '__main__':
    print(add(1,2))
    print(sub(2,1))
